import numpy as np
import glob
import os
import xml.etree.ElementTree as ET
import xml.dom.minidom


'''
第一步，将xml文件和图片重新命名
'''

# 获取文件夹中bmp图片的数量
def getDirImageNum(path):
    bmpDirImagesNum = 0
    for bmpfile in os.listdir(path):
        if os.path.splitext(bmpfile)[1] == '.jpeg':
            bmpDirImagesNum += 1
    return bmpDirImagesNum
 
# 获取文件夹中xml文件的数量
def getDirXmlNum(path):
    xmlDirXmlNum = 0
    for xmlfile in os.listdir(path):
        if os.path.splitext(xmlfile)[1] == '.xml':
            xmlDirXmlNum += 1
    return xmlDirXmlNum

   


inputpath1="./all_img_50k_fanxiu_20220622_4937_coco_0/"
inputpath2='./all_xml_50k_fanxiu_20220622_4937_coco_0/'

outpath1="./all_img_50k_fanxiu_20220622_4937_coco"
outpath2="./all_xml_50k_fanxiu_20220622_4937_coco"
if not os.path.exists(outpath1):
    os.mkdir(outpath1)
if not os.path.exists(outpath2):
    os.mkdir(outpath2)

file_name = os.listdir(inputpath2)

error = []
for item in file_name:
    # print(item)
    o_imap = inputpath1+item.split('.xml')[0]+".jpeg"
    print(o_imap)
    o_xmlp = inputpath2+item.split('.xml')[0]+".xml"
    print(o_xmlp)
    i = getDirImageNum(outpath1)  # 表示bmp文件的命名是从当前输出文件夹中的bmp文件数目开始的
    if os.path.exists(o_imap) and os.path.exists(o_xmlp):
        i = i+1
        new_name = '2019' + format(str(i), '0>7s') + '.jpeg'
        # new_name = '2019' + format(str(i)) + '.jpeg'
        dst1 = os.path.join(os.path.abspath(outpath1), new_name) # 为000001.jpeg
        os.rename(o_imap, dst1)
        dst2 = os.path.join(os.path.abspath(outpath2), '2019' + format(str(i), '0>7s') + '.xml') # 为000000.xml形式，想要的格式
        # dst2 = os.path.join(os.path.abspath(outpath2), '2019' + format(str(i)) + '.xml')    
        try:
            dom = xml.dom.minidom.parse(o_xmlp)
            root = dom.documentElement
            # 获取标签对path之间的值并赋予新值j
            # 文件夹赋值
            root.getElementsByTagName('folder')[0].firstChild.data = "VOC2007"

            # 获取标签对filename之间的值并赋予新值j
            root.getElementsByTagName('filename')[0].firstChild.data = new_name

            # 将修改后的xml文件保存,xml文件修改前后的路径
            # 打开并写入
            with open(o_xmlp, 'w') as fh:
                dom.writexml(fh)
            os.rename(o_xmlp, dst2)
            print ('converting %s to %s ...' % (o_xmlp, dst2))
        except:
            error.append(new_name)
            continue
# 如果有出错的文件，error++
print(len(error))